package com.example.demo.controller;

import com.example.demo.dto.ProductStatisticsVO;
import com.example.demo.dto.ProductVO;
import com.example.demo.utils.ExcelUtil;
import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.hssf.util.HSSFColor;
import org.apache.poi.ss.usermodel.Font;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * Created by gan on 2018/8/29 21:40.
 */

@Controller
public class DownloadController {

    private List<ProductVO> initData() {

        List<ProductVO> vos = new ArrayList<>();

        ProductStatisticsVO psv1 = new ProductStatisticsVO("甘波", new BigDecimal(99.1), new Date());
        ProductStatisticsVO psv2 = new ProductStatisticsVO("甘起", new BigDecimal(98.1), new Date());
        List<ProductStatisticsVO> psvss = new ArrayList<>();
        psvss.add(psv1);
        psvss.add(psv2);
        ProductVO pv1 = new ProductVO("中央银行", "00000000000074363999", psvss);
        vos.add(pv1);


        ProductStatisticsVO psv3 = new ProductStatisticsVO("bruce", new BigDecimal(66), new Date());
        ProductStatisticsVO psv4 = new ProductStatisticsVO("jams", new BigDecimal(88.1), new Date());
        List<ProductStatisticsVO> psvss2 = new ArrayList<>();
        psvss2.add(psv3);
        psvss2.add(psv4);
        ProductVO pv2 = new ProductVO("招商银行", "00000000000074363888", psvss2);
        vos.add(pv2);


        return vos;
    }

    @GetMapping("/downloadexcel")
    public void download(HttpServletRequest request, HttpServletResponse response) throws IOException {

        //导出的Excel头部，这个要根据自己项目改一下
        String[] headers = {"产品名字", "产品编号", "姓名", "银行", "金额", "时间"};


        List<ProductVO> datas = initData(); //查询出来的数据，根据自己项目改一下

        try {

            HSSFWorkbook workbook = ExcelUtil.export(headers, datas);
            response.setContentType("application/octet-stream");
            response.setHeader("Content-disposition", "attachment;filename=createList.xls");//默认Excel名称
            response.flushBuffer();
            workbook.write(response.getOutputStream());

        } catch (Exception e) {
            e.printStackTrace();
        }
//        // 声明一个工作薄
//        HSSFWorkbook workbook = new HSSFWorkbook();

//
//        // 生成一个表格
//        HSSFSheet sheet = workbook.createSheet();
//
//        // 设置表格默认列宽度为15个字节
//        sheet.setDefaultColumnWidth((short) 18);
//
//        //第一行设置标题
//        HSSFRow row = sheet.createRow(0);
//        for (int i = 0; i < headers.length; i++) {
//            HSSFCell cell = row.createCell(i);
//
//            //生成单元格样式
//            HSSFCellStyle cellStyle = workbook.createCellStyle();
//            //新建font实体
//            HSSFFont hssfFont = workbook.createFont();
//            //设置字体颜色
//            hssfFont.setColor(HSSFColor.BLACK.index);
//            //字体大小
//            hssfFont.setFontHeightInPoints((short) 20);
//            hssfFont.setFontName("楷体");
//            //粗体
//            hssfFont.setBoldweight(Font.BOLDWEIGHT_BOLD);
//            cellStyle.setFont(hssfFont);
//
//            cell.setCellStyle(cellStyle);
//
//            HSSFRichTextString text = new HSSFRichTextString(headers[i]);
//            cell.setCellValue(text);
//        }


//        for (int i = 0; i < datas.size(); i++) {
//            ProductVO productVO = datas.get(i);
//            HSSFCell cell = row.createCell(i);
//
//            HSSFRichTextString richString = new HSSFRichTextString(textValue);
//            HSSFFont font3 = workbook.createFont();
//            font3.setColor(HSSFColor.BLUE.index);//定义Excel数据颜色
//            richString.applyFont(font3);
//            cell.setCellValue(richString);
//
//        }

        //遍历集合数据，产生数据行
//        Iterator it = dataset.iterator();
//        int index = 0;
//        while (it.hasNext()) {
//            index++;
//            row = sheet.createRow(index);
//            User t = (User) it.next();
//            //利用反射，根据javabean属性的先后顺序，动态调用getXxx()方法得到属性值
//            Field[] fields = t.getClass().getDeclaredFields();
//            for (short i = 0; i < fields.length; i++) {
//                HSSFCell cell = row.createCell(i);
//                Field field = fields[i];
//                String fieldName = field.getName();
//                String getMethodName = "get"
//                        + fieldName.substring(0, 1).toUpperCase()
//                        + fieldName.substring(1);
//                try {
//                    Class tCls = t.getClass();
//                    Method getMethod = tCls.getMethod(getMethodName,
//                            new Class[]{});
//                    Object value = getMethod.invoke(t, new Object[]{});
//                    String textValue = null;
//
//
//                    if (value instanceof Date) {
//                        Date date = (Date) value;
//                        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
//                        textValue = sdf.format(date);
//                    } else {
//                        //其它数据类型都当作字符串简单处理
//                        textValue = value.toString();
//                    }
//
//                    HSSFRichTextString richString = new HSSFRichTextString(textValue);
//                    HSSFFont font3 = workbook.createFont();
//                    font3.setColor(HSSFColor.BLUE.index);//定义Excel数据颜色
//                    richString.applyFont(font3);
//                    cell.setCellValue(richString);
//
//                } catch (SecurityException e) {
//                    e.printStackTrace();
//                } catch (NoSuchMethodException e) {
//                    e.printStackTrace();
//                } catch (IllegalArgumentException e) {
//                    e.printStackTrace();
//                } catch (IllegalAccessException e) {
//                    // TODO Auto-generated catch block
//                    e.printStackTrace();
//                } catch (InvocationTargetException e) {
//                    e.printStackTrace();
//                }
//            }
//        }

    }

}
